Weak-to-Strong On-Policy Distillation
弱到强在线策略蒸馏:用更弱的模型提升强学生
🧭 弱到强蒸馏 · 在线策略蒸馏 ★ 深度解读
University of Maryland, College Park · Microsoft Research · MBZUAI
通过对比弱模型提取能力方向并锚定到学生基模型,构造代理教师实现弱到强蒸馏,使强学生超越弱教师
📌 工作参考价值(4 点)
- 弱教师也能提升强模型:当没有更强的教师可用时(如前沿推荐模型或Agent策略网络),用一对更小的模型(如不同规模的基模型或不同训练阶段的专家)通过logit差异提取“能力方向”,可以继续提升强模型。
- 代理教师构造方法:在在线策略蒸馏中,代理教师 = 学生基模型 + α*(正模型 − 负模型) logits。这种构造避免了分布不匹配,α控制信号强度与分布邻近度的权衡(图4:α≈1.0最优),可直接用于多领域推荐模型或Agent策略的合并。
- 低成本多专家合并:多个弱领域专家(如不同品类推荐模型)可通过各自与基础模型的对比方向求和,一次性蒸馏进同一个学生,无需顺序微调或担心灾难性遗忘(Eq.12)。
- 保留通用能力:锚定学生基模型作为参考分布,使代理教师保持与学生分布邻近,避免直接蒸馏弱教师导致的通用能力退化,这对需要保持广泛推荐能力的系统尤为重要。